import java.util.LinkedList;
import java.util.Queue;

class Solution {
//    public int[][] updateMatrix(int[][] mat) {
//        int m = mat.length, n = mat[0].length;
//        int[][] distance = new int[m][n];
//        Queue<int[]> queue = new LinkedList<>();
//        boolean[][] visited = new boolean[m][n];
//        for (int i = 0; i < m; i++) {
//            for (int j = 0; j < n; j++) {
//                if (mat[i][j] == 0) {
//                    // 将超级源点入队、修改距离数组的值
//                    queue.offer(new int[]{i, j});
//                    visited[i][j] = true;
//                    distance[i][j] = 0;
//                }
//            }
//        }
//
//        int[] dx = {1,-1,0,0};
//        int[] dy = {0,0,1,-1};
//        // 从超级源点开始进行BFS求最短距离
//        int step = 0;
//        while (!queue.isEmpty()) {
//            step++;
//            int size = queue.size();
//            while (size-- != 0) {
//                int[] arr = queue.poll();
//                for (int k = 0; k < 4; k++) {
//                    int sr = arr[0] + dx[k], sc = arr[1] + dy[k];
//                    if (sr >= 0 && sr < m && sc >= 0 && sc < n && !visited[sr][sc]) {
//                        // sr和sc下标在mat数组中对应的值只能是1
//                        distance[sr][sc] = step;
//                        // 通过该位置继续进行扩展，因此该位置也需要入队
//                        queue.offer(new int[]{sr, sc});
//                        visited[sr][sc] = true;
//                    }
//                }
//            }
//        }
//        return distance;
//    }



    public int[][] updateMatrix(int[][] mat) {
        int m = mat.length, n = mat[0].length;
        int[][] distance = new int[m][n];
        Queue<int[]> queue = new LinkedList<>();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (mat[i][j] == 0) {
                    // 将源点入队 + 设置distance数组中的值
                    queue.offer(new int[]{i, j});
                    distance[i][j] = 0;
                } else {
                    distance[i][j] = -1; // 未访问设置为-1
                }
            }
        }

        int[] dx = {1,-1,0,0};
        int[] dy = {0,0,1,-1};

        while (!queue.isEmpty()) {
            // 这里也可以加上size
            int[] arr = queue.poll();
            for (int k = 0; k < 4; k++) {
                int sr = arr[0] + dx[k], sc = arr[1] + dy[k];
                // 判断是否越界，是否未访问
                if (sr >= 0 && sr < m && sc >= 0 && sc < n && distance[sr][sc] == -1) {
                    distance[sr][sc] = distance[arr[0]][arr[1]] + 1;
                    queue.offer(new int[]{sr, sc});
                }
            }
        }
        return distance;
    }
}